package zby;

import com.sun.javafx.binding.StringFormatter;

import java.text.Format;
import java.util.*;

public class LeetCode_15 {

    /**
     * 15. 三数之和
     * <p>
     * 给你一个包含 n 个整数的数组 nums，判断 nums 中是否存在三个元素 a，b，c ，使得 a + b + c = 0 ？请你找出所有满足条件且不重复的三元组。
     * <p>
     * 注意：答案中不可以包含重复的三元组。
     *
     * @param args
     */
    public static void main(String[] args) {
        LeetCode_15 code = new LeetCode_15();
        int[] nums = {-1, 0, 1, 2, -1, -4};
        List<List<Integer>> lists = code.threeSum3(nums);
        System.out.println(lists);
    }

    /**
     * 19 ms	41.2 MB
     *
     * @param nums
     * @return
     */
    private List<List<Integer>> threeSum3(int[] nums) {
        List result = new ArrayList<>(nums.length / 3);
        Arrays.sort(nums);
        int v, i, j;
        for (int k = 0; k < nums.length; k++) {
            if (nums[k] > 0) break;
            if (k > 0 && nums[k] == nums[k - 1]) continue;
            i = k + 1;
            j = nums.length - 1;
            while (i < j) {
                v = nums[k] + nums[i] + nums[j];
                if (v == 0) {
                    result.add(new int[]{nums[k], nums[i++], nums[j--]});
                    while (i < j && nums[i] == nums[i - 1]) i++;
                    while (i < j && nums[j] == nums[j + 1]) j--;
                } else if (v < 0) {
                    i++;
                } else {
                    j--;
                }
            }
        }
        return result;
    }

    /**
     * 18 ms	42.6 MB
     *
     * @param nums
     * @return
     */
    private List<List<Integer>> threeSum2(int[] nums) {
        List<List<Integer>> result = new ArrayList<>();
        if (nums == null || nums.length == 0) return result;
        Arrays.sort(nums);
        for (int k = 0; k < nums.length; k++) {
            if (nums[k] > 0) break;
            if (k > 0 && nums[k] == nums[k - 1]) continue;
            int i = k + 1, j = nums.length - 1;
            int target = -nums[k];
            while (i < j) {
                if (nums[i] + nums[j] > target) j--;
                else if (nums[i] + nums[j] < target) i++;
                else {
                    ArrayList<Integer> temp = new ArrayList<>();
                    temp.add(nums[i]);
                    temp.add(nums[j]);
                    temp.add(nums[k]);
                    result.add(temp);
                    while (i < j && nums[i] == nums[i + 1]) i++;
                    while (i < j && nums[j] == nums[j - 1]) j--;
                    i++;
                    j--;
                }
            }
        }
        return result;
    }

    /**
     * 31 ms	42.4 MB
     *
     * @param nums
     * @return
     */
    private List<List<Integer>> threeSum1(int[] nums) {
        List<List<Integer>> result = new ArrayList<>();
        Arrays.sort(nums);
        for (int k = 0; k < nums.length - 2; k++) {
            if (nums[k] > 0) break;
            if (k > 0 && nums[k - 1] == nums[k]) continue;
            int i = k + 1, j = nums.length - 1;
            while (i < j) {
                int s = nums[i] + nums[j] + nums[k];
                if (s < 0) {
                    while (i < j && nums[i] == nums[++i]) ;
                } else if (s > 0) {
                    while (i < j && nums[j] == nums[--j]) ;
                } else {
                    result.add(new ArrayList<>(Arrays.asList(nums[i], nums[j], nums[k])));
                    while (i < j && nums[i] == nums[++i]) ;
                    while (i < j && nums[j] == nums[--j]) ;
                }
            }
        }
        return result;
    }


    /**
     * 超出时间限制	N/A	N/A
     *
     * @param nums
     * @return
     */
    public List<List<Integer>> threeSum(int[] nums) {
        Set<List<Integer>> result = new HashSet<>();
        for (int i = 0; i < nums.length - 2; i++) {
            for (int j = i + 1; j < nums.length - 1; j++) {
                for (int k = j + 1; k < nums.length; k++) {
                    if (nums[i] + nums[j] + nums[k] == 0) {
                        ArrayList<Integer> list = new ArrayList<>(Arrays.asList(nums[i], nums[j], nums[k]));
                        Collections.sort(list);
                        result.add(list);
                    }
                }
            }
        }
        return new ArrayList<>(result);
    }
}
